package cn.ciis.basic.model;

import cn.ciis.basic.config.CiisConfig;
import cn.ciis.basic.kit.CiisKit;
import com.jfinal.kit.LogKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

/**
 * Model、BaseModel、_MappingKit 生成器
 *
 * @author 冉志林
 */
public class GeneratorModel {

  /** 部分功能使用 Db + Record 模式实现，无需生成 model 的 table 在此配置 */
  private static String[] excludedTable = {
    // 暂不实现该功能
    "sys_login_log",
    "sys_sensitive_words",
    "sys_upload_counter",
    "sys_account_role",
    "sys_role_permission",
    "sys_action_notes"
  };

  /** 重用 JFinalClubConfig 中的数据源配置，避免冗余配置 */
  public static DataSource getDataSource() {
    DruidPlugin druidPlugin = CiisConfig.getDruidPlugin();
    druidPlugin.start();
    return druidPlugin.getDataSource();
  }

  public void gen() {
    // base model 所使用的包名
    String baseModelPackageName = "cn.ciis.basic.model.base";
    // base model 文件保存路径
    String baseModelOutputDir =
        CiisKit.getWebRoot() + "/src/main/java/cn/ciis/basic/model/base";
    LogKit.info("输出路径：" + baseModelOutputDir);
    // model 所使用的包名 (MappingKit 默认使用的包名)
    String modelPackageName = "cn.ciis.basic.model";
    // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
    String modelOutputDir = baseModelOutputDir + "/..";

    // 创建生成器
    Generator gen =
        new Generator(
            getDataSource(),
            baseModelPackageName,
            baseModelOutputDir,
            modelPackageName,
            modelOutputDir);

    // 设置数据库方言
    gen.setDialect(new MysqlDialect());

    // 设置是否生成字段备注
    gen.setGenerateRemarks(true);

    // 添加不需要生成的表名
    for (String table : excludedTable) {
      gen.addExcludedTable(table.trim());
    }

    // 设置是否在 Model 中生成 dao 对象
    gen.setGenerateDaoInModel(false);
    // 设置是否生成字典文件
    gen.setGenerateDataDictionary(false);
    // 设置需要被移除的表名前缀用于生成modelName
    gen.setRemovedTableNamePrefixes("ciis_","sys_");
    gen.generate();
  }
}
